#
# 6.863 -- Final Project
#
# Author: Andreea Bodnari
# Contact: andreeab@mit.edu
# 5/9/11
#
#

from src.train.ngramTrain import NgramTrain

if __name__ == '__main__':
    '''
    Test cases for the n-gram train functionality
    '''
    trainModel = NgramTrain(3, None, "treebank")
    
    print trainModel.wordSequenceGeneration(7)
    print trainModel.wordGeneration(None, "about")
    
